package com.project.mq.mqserver.core;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.project.mq.common.ConsumerEnv;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * msg => Message
 * 这个类表示存储消息的队列
 */
@Data
public class MSGQueue {
    //队列的名字. 相当于队列的⾝份标识.(唯一的)
    private String name;
    //交换机是否要持久化存储. true 为持久化, false 不持久化.
    private boolean durable = false;
    //独占(排他), 如果为true,队列只能被⼀个消费者使⽤.为false都能用,暂时不实现
    private boolean exclusive = false;
    //使⽤完毕后是否⾃动删除.预留字段, 暂时未使⽤.
    private boolean autoDelete = false;
    //交换机的其他参数属性. 预留字段, 暂时未使⽤.
    private Map<String, Object> arguments = new HashMap<>();
    // 当前队列都有哪些消费者订阅了。
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();
    // 轮询序号 记录当前去到了的几个消费者,方便实现轮询
    private AtomicInteger consumerSeq = new AtomicInteger(0);
    // 省略 getter setter
    public String getArguments() {
        // 是把当前的 arguments 参数，从 Map 转成 String（JSON）
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        //如果代码真异常了,就返回一个空的json字符串就可以
        return "{}";
    }

    //这个方法,是从数据库读数据之后,构造Exchange对象,自动调用到
    public void setArguments(String argumentsJson) {
        //把参数中的argumentsJson按照JSON格式解析,转成上述的Map对象
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String,Object>>() {});
        } catch (JsonProcessingException e) {
            throw new RuntimeException(e);
        }
    }

    // 在这里针对 arguments，再提供一组 getter setter ，用来去更方便的获取/设置这里的键值对.
    //在代码内部测试使用
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }
    //订阅者的删除我们暂时不考虑

    //挑选一个订阅者,用来处理当前的信息,按照轮询的方式
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.size() == 0) {
            return null;
        }
        //计算当前要去的元素下表
        int index = consumerSeq.get() % consumerEnvList.size();
        //自增,方便去下一个下标
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }
}
